import os
import uuid
import shutil
class MyFiles:
    def __init__(self,upload_dir='./static/upload/'):
        self.up_dir = upload_dir

    #创建文件夹  不能穿件 路径有问题
    def mk_dir(self,userid):
        #判断是否存在
        path = self.up_dir+str(userid)+'/'
        if not os.path.exists(path):
            os.mkdir(path)

    #文件重命名  完成 
    def rename(self,file):
        filename=uuid.uuid4().hex+os.path.splitext(file)[1] 
        return filename

    #查找指定目录下的文件
    def show_all(self,userid):
        files = os.listdir(self.up_dir+str(userid)+'/')
        return files

    #删除一个文件
    def remove_file(self,userid,file):
        path = self.up_dir+str(userid)+'/'+file
        os.remove(path)


    #删除整个文件文件
    def remove_dir(self,userid):
        #删除一个文件
        # os.rmdir(self.up_dir+str(userid))

        #删除一个文件树 就是整个目录
        path = self.up_dir + str(userid)+'/'
        #判断有没有整个路径
        if os.path.exists(path):
            #删除文件夹及下边所有文件
            shutil.rmtree(path)


myf = MyFiles()

# myf.mk_dir(2)
# print(myf.rename('123.jpg'))
# print(myf.show_all(23))
# myf.remove_dir(22)